#include "../server.hpp"
#include <fstream>
#include <regex>
#include <assert.h>
#include <sys/stat.h>

#define DEFALT_TIMEOUT 10
std::unordered_map<int, std::string> _status_msg = {
    {100,  "Continue"},
    {101,  "Switching Protocol"},
    {102,  "Processing"},
    {103,  "Early Hints"},
    {200,  "OK"},
    {201,  "Created"},
    {202,  "Accepted"},
    {203,  "Non-Authoritative Information"},
    {204,  "No Content"},
    {205,  "Reset Content"},
    {206,  "Partial Content"},
    {207,  "Multi-Status"},
    {208,  "Already Reported"},
    {226,  "IM Used"},
    {300,  "Multiple Choice"},
    {301,  "Moved Permanently"},
    {302,  "Found"},
    {303,  "See Other"},
    {304,  "Not Modified"},
    {305,  "Use Proxy"},
    {306,  "unused"},
    {307,  "Temporary Redirect"},
    {308,  "Permanent Redirect"},
    {400,  "Bad Request"},
    {401,  "Unauthorized"},
    {402,  "Payment Required"},
    {403,  "Forbidden"},
    {404,  "Not Found"},
    {405,  "Method Not Allowed"},
    {406,  "Not Acceptable"},
    {407,  "Proxy Authentication Required"},
    {408,  "Request Timeout"},
    {409,  "Conflict"},
    {410,  "Gone"},
    {411,  "Length Required"},
    {412,  "Precondition Failed"},
    {413,  "Payload Too Large"},
    {414,  "URI Too Long"},
    {415,  "Unsupported Media Type"},
    {416,  "Range Not Satisfiable"},
    {417,  "Expectation Failed"},
    {418,  "I'm a teapot"},
    {421,  "Misdirected Request"},
    {422,  "Unprocessable Entity"},
    {423,  "Locked"},
    {424,  "Failed Dependency"},
    {425,  "Too Early"},
    {426,  "Upgrade Required"},
    {428,  "Precondition Required"},
    {429,  "Too Many Requests"},
    {431,  "Request Header Fields Too Large"},
    {451,  "Unavailable For Legal Reasons"},
    {501,  "Not Implemented"},
    {502,  "Bad Gateway"},
    {503,  "Service Unavailable"},
    {504,  "Gateway Timeout"},
    {505,  "HTTP Version Not Supported"},
    {506,  "Variant Also Negotiates"},
    {507,  "Insufficient Storage"},
    {508,  "Loop Detected"},
    {510,  "Not Extended"},
    {511,  "Network Authentication Required"}
};

std::unordered_map<std::string, std::string> _mime_msg = {
    {".aac",        "audio/aac"},
    {".abw",        "application/x-abiword"},
    {".arc",        "application/x-freearc"},
    {".avi",        "video/x-msvideo"},
    {".azw",        "application/vnd.amazon.ebook"},
    {".bin",        "application/octet-stream"},
    {".bmp",        "image/bmp"},
    {".bz",         "application/x-bzip"},
    {".bz2",        "application/x-bzip2"},
    {".csh",        "application/x-csh"},
    {".css",        "text/css"},
    {".csv",        "text/csv"},
    {".doc",        "application/msword"},
    {".docx",       "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    {".eot",        "application/vnd.ms-fontobject"},
    {".epub",       "application/epub+zip"},
    {".gif",        "image/gif"},
    {".htm",        "text/html"},
    {".html",       "text/html"},
    {".ico",        "image/vnd.microsoft.icon"},
    {".ics",        "text/calendar"},
    {".jar",        "application/java-archive"},
    {".jpeg",       "image/jpeg"},
    {".jpg",        "image/jpeg"},
    {".js",         "text/javascript"},
    {".json",       "application/json"},
    {".jsonld",     "application/ld+json"},
    {".mid",        "audio/midi"},
    {".midi",       "audio/x-midi"},
    {".mjs",        "text/javascript"},
    {".mp3",        "audio/mpeg"},
    {".mpeg",       "video/mpeg"},
    {".mpkg",       "application/vnd.apple.installer+xml"},
    {".odp",        "application/vnd.oasis.opendocument.presentation"},
    {".ods",        "application/vnd.oasis.opendocument.spreadsheet"},
    {".odt",        "application/vnd.oasis.opendocument.text"},
    {".oga",        "audio/ogg"},
    {".ogv",        "video/ogg"},
    {".ogx",        "application/ogg"},
    {".otf",        "font/otf"},
    {".png",        "image/png"},
    {".pdf",        "application/pdf"},
    {".ppt",        "application/vnd.ms-powerpoint"},
    {".pptx",       "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    {".rar",        "application/x-rar-compressed"},
    {".rtf",        "application/rtf"},
    {".sh",         "application/x-sh"},
    {".svg",        "image/svg+xml"},
    {".swf",        "application/x-shockwave-flash"},
    {".tar",        "application/x-tar"},
    {".tif",        "image/tiff"},
    {".tiff",       "image/tiff"},
    {".ttf",        "font/ttf"},
    {".txt",        "text/plain"},
    {".vsd",        "application/vnd.visio"},
    {".wav",        "audio/wav"},
    {".weba",       "audio/webm"},
    {".webm",       "video/webm"},
    {".webp",       "image/webp"},
    {".woff",       "font/woff"},
    {".woff2",      "font/woff2"},
    {".xhtml",      "application/xhtml+xml"},
    {".xls",        "application/vnd.ms-excel"},
    {".xlsx",       "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    {".xml",        "application/xml"},
    {".xul",        "application/vnd.mozilla.xul+xml"},
    {".zip",        "application/zip"},
    {".3gp",        "video/3gpp"},
    {".3g2",        "video/3gpp2"},
    {".7z",         "application/x-7z-compressed"}
};

class Util{
public:
    // 字符串分割函数 -- 将 src 字符串按照 sep 字符分割得到的字串放入 arry 中, 返回字串数量
    static size_t Split(const std::string &src, const std::string &sep, std::vector<std::string> *array){
        array->clear(); // 清除旧数据
        if(sep.empty()) return 0;
        
        size_t offset = 0; // 用偏移量来查找
        while(offset < src.size()){
            size_t pos = src.find(sep, offset);
            // 没找到
            if(pos == std::string::npos){
                // 把剩余的部分当作一个字串
                array->emplace_back(src.substr(offset));
                break;
            }

            // 空字串不要
            if(pos == offset){
                offset += sep.size();
                continue;
            }
            array->emplace_back(src.substr(offset, pos - offset));
            offset = pos + sep.size();
        }

        return array->size();
    }

    // 读取内容/写入数据 向文件
    // 读取文件的所有内容把读取的内容放到 buffer 中
    static bool ReadFile(const std::string& filename, std::string* buf) {
        // 避免 string 设定为 nullptr, 这样会导致 未定义行为
        if (!buf) {
            LOG_ERROR("Invalid buffer pointer.");
            return false;
        }

        std::ifstream ifs(filename, std::ios::binary);
        if (!ifs.is_open()) {
            LOG_ERROR("Failed to open file: %s", filename.c_str());
            return false;
        }

        // ifs.seekg(0, ifs.end);      // 跳转读写位置到末尾
        // size_t fsize = ifs.tellg(); // 获取当前读写位置相对于起始位置的偏移量，从末尾偏移刚好就是文件大小
        // ifs.seekg(0, ifs.beg);      // 跳转到起始位置
        // buf->resize(fsize);         // 开辟文件大小的空间
        // ifs.read(&(*buf)[0], fsize);

        ifs.seekg(0, ifs.end); // 跳转读写位置到文件末尾
        // 获取当前读写位置相对于起始位置的偏移量，从末尾偏移刚好就是文件大小
        // 避免 size_t 得到 负数转化为极大值
        std::ifstream::pos_type pos = ifs.tellg();
        if (pos == std::ifstream::pos_type(-1)) {
            LOG_ERROR("Failed to get file size.");
            return false;
        }
        size_t fsize = static_cast<size_t>(pos);

        ifs.seekg(0, ifs.beg); // 跳转起始位置
        buf->resize(fsize);

        // gcount 是检查是否读取完整
        ifs.read(&(*buf)[0], fsize);
        if(!ifs.good()){
            LOG_ERROR("Read file FAILED: %s", filename.c_str());
            ifs.close();
            return false;
        }
        if (ifs.gcount() != static_cast<std::streamsize>(fsize)) {
            LOG_ERROR("Incomplete read from file.");
            ifs.close();
            return false;
        }
        ifs.close(); // 记得关闭
        return true;
    }

    static bool WriteFile(const std::string &filename, const std::string &buf, bool append = false){
        std::ios::openmode mode = std::ios::binary;
        if(append) mode |= std::ios::app;
        else mode |= std::ios::trunc; // trunc 表示之前内容不要, 重新写
        std::ofstream ofs(filename, mode);

        if(!ofs.is_open()){
            LOG_ERROR("Failed to open file: %s", filename.c_str());
            ofs.close();
            return false;
        }
        ofs.write(buf.c_str(), buf.size());
        if(!ofs){
            LOG_ERROR("Failed to write to file: %s", filename.c_str());
            ofs.close();
            return false;
        }
        ofs.close();
        return true;
    }

    // URL 编码/解码
    static std::string UrlEncode(const std::string& url, bool convert_space_to_plus){
        std::string res;
        for(char c: url){
            // 绝对不编码字符 isalnum: 数字/字母  未保留字符
            if (std::isalnum(c) || c == '-' || c == '_' || c == '.' || c == '~' ) {
                res += c;
            } else if (c == ' ' && convert_space_to_plus) {
                res += '+';
            } else {
                char tmp[4]; // 不需要初始化为 {0}，snprintf 会处理
                snprintf(tmp, sizeof(tmp), "%%%02X", static_cast<unsigned char>(c)); // 将 字符转化为 %16进制
                res += tmp;
            }
        }
        return res;
    }

    // 16 进制转化
    static int HEXTOI(char c) {
        if (isdigit(c)) return c - '0';
        else if (c >= 'a' && c <= 'f') return c - 'a' + 10;
        else if (c >= 'A' && c <= 'F') return c - 'A' + 10;
        return -1; // 表示非法字符
    }

    // URL 解码
    static std::string UrlDecode(const std::string& url, bool convert_space_to_plus){
        // 遇到了%，则将紧随其后的2个字符，转换为数字，第一个数字左移4位，然后加上第二个数字  + -> 2b  %2b->2 << 4 + 11
        std::string res;
        // 预估结果字符串大小，可以减少内存重新分配
        res.reserve(url.length()); // 解码后通常不会比原字符串更长

        for(int i = 0; i < url.size(); ++i){
            if(url[i] == '+' && convert_space_to_plus){
                res += ' ';
                continue;
            }
            else if(url[i] == '%' && (i + 2) < url.size()){
                int v1 = HEXTOI(url[i + 1]), v2 = HEXTOI(url[i + 2]);
                 // 只有当 v1 和 v2 都合法时才进行解码
                if(v1 != -1 && v2 != -1){ // <-- 修正逻辑：必须是 && (都有效)
                    res += static_cast<char>(v1 * 16 + v2); // 解码并添加到结果
                    i += 2; // 跳过已处理的两个十六进制字符
                }
                else{
                    // 如果 '%xx' 序列非法（例如 '%' 后跟非十六进制字符或字符不足），
                    // 则将 '%' 字符原样保留在结果中
                    res += '%';
                    // 此时不跳过 i，让 i+1 和 i+2 在下一次循环被当作普通字符处理
                    // 或者你可以选择直接跳过 i+1 让其原样保留
                    // 如果你想让非法字符也原样保留，可以这样做：
                    // res += url[i+1];
                    // res += url[i+2];
                    // i += 2;
                    // 但通常情况下，只保留 '%' 更符合预期
                }
            }
            else res += url[i];
        }
        return res;
    }

    // 响应状态码描述信息获取
    static std::string StatuDesc(int statu){
        auto it = _status_msg.find(statu);
        if(it != _status_msg.end()){
            return it->second;
        }
        return "UnKnow";
    }

    // 根据文件后缀名获取文件 mime
    static std::string ExtMime(const std::string &filename){
        // 比如: a.b.txt 先获取文件扩展名
        size_t pos = filename.find_last_of('.');
        if(pos == std::string::npos){
            return "application/octet-stream";
        }

        // 根据扩展名获取 mime
        std::string ext = filename.substr(pos);
        auto it = _mime_msg.find(ext);
        if(it == _mime_msg.end()){
            return "application/octet-stream";
        }
        return it->second;
    }

    // 判断一个文件是否是 目录/普通文件
    static bool IsDirectory(const std::string &filename){
        struct stat st;
        if(stat(filename.c_str(), &st) < 0){
            return false;
        }
        return S_ISDIR(st.st_mode);
    }
    static bool IsRegular(const std::string &filename){
        struct stat st;
        if(stat(filename.c_str(), &st) < 0){
            return false;
        }
        return S_ISREG(st.st_mode);
    }

    // http 请求资源路径的有效性判断

    // /index.html  --- 前边的/叫做相对根目录  映射的是某个服务器上的子目录
    // 想表达的意思就是，客户端只能请求相对根目录中的资源，其他地方的资源都不予理会
    // /../login, 这个路径中的..会让路径的查找跑到相对根目录之外，这是不合理的，不安全的
    static bool ValidPath(const std::string &path){
        // 思想: 按照 / 进行路径分割, 根据有多少子目录计算目录深度的层数, 注意这个深度不能 <0
        if(path.empty()) return false;

        std::vector<std::string> subdir;
        Split(path, "/", &subdir);
        int level = 0;
        for(auto &dir: subdir){
            if(dir.empty() || dir == ".") continue; // 忽略空目录和当前目录
            if(dir == ".."){
                level--;
                // 任意一层走出相对根目录，就认为有问题
                if(level < 0)  return false;
            }
            else level++; // 普通目录进入下一级
        }
        return true;
    }
};


// Case-Insentive: 不区分大小写, 自定义大小写不敏感的哈希和比较器
struct CaseInsensitiveHash {
    size_t operator()(const std::string& s) const{
        std::hash<std::string> h;
        std::string lower = s;
        // 转为小写
        std::transform(lower.begin(), lower.end(), lower.begin(), ::tolower);
        
        return h(lower); // 小写字符串的哈希
    }
};

struct CaseInsensitiveEqual {
    bool operator()(const std::string& a, const std::string& b) const {
        // std::equal 的带谓词版本接受 3个迭代器参数 + 1个谓词 ，而不是4个迭代器参数
        // template<class InputIt1, class InputIt2, class BinaryPredicate>
        // bool equal(InputIt1 first1, InputIt1 last1, InputIt2 first2, BinaryPredicate p);
        
        return std::equal(a.begin(), a.end(), b.begin(),
            [](char c1, char c2) {return tolower(c1) == tolower(c2); });
    }
};

using HeaderMap = std::unordered_map<std::string, std::string, CaseInsensitiveHash, CaseInsensitiveEqual>;


class HttpRequest{
public:
    void ReSet(){
        _method.clear();
        _path.clear();
        _version = "HTTP/1.1";
        _body.clear();

        // 因为 match 无 clear 函数, 因此通过 swap 与空 smatch 交换，实现重置效果
        std::smatch match;
        _matches.swap(match);
        _headers.clear();
        _params.clear();
    }

    HttpRequest(): _version("HTTP/1.1") {} 

    // 设置头部字段
    void SetHeader(const std::string &key, const std::string &val){
        // operator vs insert -- 后者 key 对应 val 是无法覆盖的
        // _headers.insert(std::make_pair(key, val));
        _headers[key] = val;
    }
    // 判断是否存在指定头部字段
    bool HasHeader(const std::string &key) const{
        return _headers.find(key) != _headers.end();
    }
    // 获取指定头部字段的值
    std::string GetHeader(const std::string &key) const{
        auto it = _headers.find(key);
        return (it != _headers.end()) ? it->second : "";
    }

    // 插入查询字符串
    void SetParam(const std::string &key, const std::string &val){
        _params[key] = val;
    }
    //判断是否有某个指定的查询字符串
    bool HasParam(const std::string &key) const {
        return _params.find(key) != _params.end();
    }
    // 获取指定的查询字符串
    std::string GetParam(const std::string &key) const{
        auto it = _params.find(key);
        return (it != _params.end()) ? it->second : "";
    }

    // 获取正文长度
    size_t ContentLength() const{
        if (!HasHeader("Content-Length")) return 0;
        std::string clen = GetHeader("Content-Length");
        try{
            return static_cast<size_t>(std::stoul(clen));
        }catch(...){
            std::cerr << "Invalid Content-Length: " << clen << std::endl;
            return 0; 
        }
    }

    // 判断是否为短连接
    bool Close() const{
        // 没有 Connection 字段, 或者有 Connection 但是值是 close, 则都是短连接, 否则都是长连接
        
        // if(HasHeader("Connection") && GetHeader("Connection") == "keep-alive"){
        //     return false;
        // }
        // return true;
        return GetHeader("Connection") != "keep-alive"; // 默认短连接
    }

    // 调试输出
    std::string ToString() const {
        std::string str;
        str += _method + " " + _path + " " + _version + "\r\n";
        
        // 注意: auto &[k, v] 是 C++17 引入的结构化绑定语法
        // for (const auto &[k, v] : _headers) {
        //     str += k + ": " + v + "\r\n";
        // }

        for(const auto &p: _headers){
            str += p.first +": " + p.second + "\r\n";
        }
        str += "\r\n" + _body;
        return str;
    }

public: 
    std::string _method;    // 请求方法
    std::string _path;      // 资源路径
    std::string _version;   // 协议版本
    std::string _body;      // 请求正文
    std::smatch _matches;   // 资源路径的正则提取数据
    
    //std::unordered_map<std::string, std::string> _headers;  // 头部字段
    HeaderMap _headers;  // 头部字段（大小写不敏感）

    std::unordered_map<std::string, std::string> _params;   // 查询字符串
};



class HttpResponse{
public:
    HttpResponse(int status = 200): _redirect_flag(false), _status(status){}

    void ReSet(){
        _status = 200;
        _redirect_flag = false;
        _body.clear();
        _redirect_url.clear();
        _headers.clear();
    }

    void SetHeader(const std::string &key, const std::string &val){
        // _headers.insert(std::make_pair(key, val));
        _headers[key] = val;
    }
    bool HasHeader(const std::string &key){
        return _headers.find(key) != _headers.end();
    }
    std::string GetHeader(const std::string &key) const {
        auto it = _headers.find(key);
        return it != _headers.end() ? it->second : "";
    }

    void SetContent(const std::string &body, const std::string &type = "text/html"){
        _body = body;
        SetHeader("Content-Type", type);
    }

    // 设置重定向
    void SetRedirect(const std::string &url, int status = 302){
        _status = status;
        _redirect_flag = true;
        _redirect_url = url;
    }
    // 判断是否为 短连接
    bool Close(){
        return GetHeader("Connection") != "keep-alive";
    }
public:
    int _status;
    bool _redirect_flag; // 重定向
    std::string _body;
    std::string _redirect_url;

    // std::unordered_map<std::string, std::string> _headers;
    HeaderMap _headers;
};


typedef enum {
    RECV_HTTP_ERROR,
    RECV_HTTP_LINE,
    RECV_HTTP_HEAD,
    RECV_HTTP_BODY,
    RECV_HTTP_OVER
}HttpRecvStatus;
#define MAX_LINE 8192

class HttpContext {
private:
    // 解析请求行
    bool ParseHttpLine(const std::string &line){
        std::smatch matches;
        std::regex e(R"((GET|HEAD|POST|PUT|DELETE) ([^? ]+)(?:\?([^ ]+))? (HTTP/1\.[01])\r?\n)");
        if(!std::regex_match(line, matches, e)){
            _resp_status = 400; // BAD REQUEST
            _recv_status =RECV_HTTP_ERROR;
            return false;
        }

        // 完整匹配: GET /Island/login?user=xiao&pass=123456 HTTP/1.1
        // HTTP 方法: GET meches[1]
        // 路径: /Island/login meches[2]
        // 查询参数: user=xiao&pass=123456 meches[3]
        // 协议版本: HTTP/1.1 meches[2]

        // 1. 请求方法获取
        _request._method = matches[1];
        // 将请求方法转为 大写 提高兼容性
        std::transform(_request._method.begin(), _request._method.end(), _request._method.begin(), ::toupper);

        // 2. 资源路径获取 -- 由于 URL 中的内容都是编码过的, 故需要先解码 不进行空格 + 转化
        _request._path = Util::UrlDecode(matches[2], false);
        // 3. 协议版本获取
        _request._version = matches[4];

        // 4. 查询字符串的获取和处理
        if(matches[3].matched){
            std::vector<std::string> query_string_arry;
            std::string query_string = matches[3];
            
            // 查询字符串的格式 key=val&key=val....., 先以 & 符号进行分割，得到各个字串
            Util::Split(query_string, "&", &query_string_arry);
            // 针对各个字串，以 = 符号进行分割，得到key 和val， 得到之后也需要进行URL解码
            for(auto &str: query_string_arry){
                size_t pos = str.find("=");
                if(pos == std::string::npos){ // 未找到
                    _resp_status = 400;
                    _recv_status = RECV_HTTP_ERROR;
                    return false;
                }
                // 取出 key val -- + -> 空格
                std::string key = Util::UrlDecode(str.substr(0, pos), true);  
                std::string val = Util::UrlDecode(str.substr(pos + 1), true);
                _request.SetParam(key, val);
            }
        }
        return true;
    }

    // 接收一行数据 -- 请求行
    bool RecvHttpLine(Buffer *buf){
        if(_recv_status != RECV_HTTP_LINE) return false;
        // 1. 获取一行数据
        std::string line = buf->GetLineAndPop();
        // 2. 情况：缓冲区数据不足一行 | 获取的一行数据很大
        if(line.size() == 0){
            // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，如果很长了都不足一行，这是有问题的
            if(buf ->ReadAbleSize() > MAX_LINE){
                _resp_status = 414;  // URI Too Long
                _recv_status = RECV_HTTP_ERROR;
                return false;
            }
            // 不足一行, 数据不多, 因此继续等待新数据
            return true;
        }else if(line.size() > MAX_LINE){
            _resp_status = 414;  // URI Too Long
            _recv_status = RECV_HTTP_ERROR;
            return false;
        }
        if(!ParseHttpLine(line)) return false;

        // 首行处理完毕, 进入头部获取阶段
        _recv_status = RECV_HTTP_HEAD;
        return true;
    }

    // 接收头部
    bool RecvHttpHead(Buffer *buf){
        if(_recv_status != RECV_HTTP_HEAD) return false;
        // 一行一行取出. 直到空行为止, 头部格式 key: val\r\n key: val\r\n
        while(true){
            std::string line = buf->GetLineAndPop();
            // 2. 情况：缓冲区数据不足一行 | 获取的一行数据很大
            if(line.size() == 0){
                // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，如果很长了都不足一行，这是有问题的
                if(buf ->ReadAbleSize() > MAX_LINE){
                    _resp_status = 414;  // URI Too Long
                    _recv_status = RECV_HTTP_ERROR;
                    return false;
                }
                // 不足一行, 数据不多, 因此继续等待新数据
                return true;
            }else if(line.size() > MAX_LINE){
                _resp_status = 414;  // URI Too Long
                _recv_status = RECV_HTTP_ERROR;
                return false;
            }
            if(line == "\r" || line == "\n" || line == "\r\n") break;
            if(!ParseHttpHead(line)) return false;
        }
        // 头部获取结束, 进入正文获取阶段
        _recv_status = RECV_HTTP_BODY;
        return true;
    }

    bool ParseHttpHead(std::string &line){
        // key: val\r\n key: val\r\n ...
        // 末尾取出换行| 回车符
        if (line.back() == '\n') line.pop_back();//末尾是换行则去掉换行字符
        if (line.back() == '\r') line.pop_back();//末尾是回车则去掉回车字符
        size_t pos = line.find(":");
        if(pos == std::string::npos){
            _recv_status = RECV_HTTP_ERROR;
            _resp_status = 400;
            return false;
        }
        std::string key = line.substr(0, pos), val = line.substr(pos + 1);

        // 去除 val 前后空格
        val.erase(val.begin(), std::find_if(val.begin(), val.end(), [](char c){ return !std::isspace(c); }));
        val.erase(std::find_if(val.rbegin(), val.rend(), [](char c){ return !std::isspace(c); }).base(), val.end());

        _request.SetHeader(key, val);
        return true;
    }

    // 接收正文
    bool RecvHttpBody(Buffer* buf){
        if(_recv_status != RECV_HTTP_BODY) return false;
        // 1. 获取正文长度
        size_t content_len = _request.ContentLength();
        if(content_len == 0){
            // 无正文, 请求接收解析完毕
            _recv_status = RECV_HTTP_OVER;
            return true;
        }
        // 2. 当前已经接收了多少正文,其实就是往  _request._body 中放了多少数据了
        size_t real_len = content_len - _request._body.size(); // 实际还需要接收的正文长度
        // 未处理正文长度不足的情况 ：如果 buf->ReadableSize() < real_len，应继续等待数据
        size_t avail_len = buf->ReadAbleSize();

        //3. 接收正文放到body中，但是也要考虑当前缓冲区中的数据，是否是全部的正文
        //  3.1 缓冲区中数据，包含了当前请求的所有正文，则取出所需的数据
        if (avail_len >= real_len) {
            _request._body.append(buf->ReadPos(), real_len);
            buf->MoveReadOffset(real_len);
            _recv_status = RECV_HTTP_OVER;
            return true;
        }
        //  3.2 缓冲区中数据，无法满足当前正文的需要，数据不足，取出数据，然后等待新数据到来
        _request._body.append(buf->ReadPos(), avail_len);
        buf->MoveReadOffset(avail_len);
        return true;
    }

public:
    HttpContext():_resp_status(200), _recv_status(RECV_HTTP_LINE){}
    void ReSet(){
        _resp_status = 200; 
        _recv_status = RECV_HTTP_LINE;
        _request.ReSet();
    }
    int RespStatus(){return _resp_status;}
    HttpRecvStatus RecvStatus() {return _recv_status;}
    HttpRequest &Request(){return _request;}

    // 接收并解析 Http 请求
    void RecvHttpRequest(Buffer *buf){
        // 不同的状态，做不同的事情，但是这里不要break， 因为处理完请求行后，应该立即处理头部，而不是退出等新数据
        switch(_recv_status){
            case RECV_HTTP_LINE:
                if (!RecvHttpLine(buf)) return;
                // fall-through
            case RECV_HTTP_HEAD:
                if (!RecvHttpHead(buf)) return;
                // fall-through
            case RECV_HTTP_BODY:
                if (!RecvHttpBody(buf)) return;
                break;
            default:
                return;
        }
    }

private:
    int _resp_status;                // 响应状态码
    HttpRecvStatus _recv_status;    // 当前接收及解析的阶段状态
    HttpRequest _request;           // 已经解析得到的请求信息
};

class HttpServer
{
private:
    using Handler = std::function<void(const HttpRequest &, HttpResponse *)>;
    using Handlers = std::vector<std::pair<std::regex, Handler>>;
    void ErrorHandler(const HttpRequest, HttpResponse *rsp){
        // 1. 组织一个错误展示页面
        std::string body = "<html>";
        body += "<head><meta http-equiv='Content-Type' content='text/html;charset=utf-8'></head>";
        body += "<body><h1>";
        body += std::to_string(rsp->_status);
        body += " ";
        body += Util::StatuDesc(rsp->_status);
        body += "</h1></body></html>";
        // 2. 将页面数据当作响应正文, 放入 rsp 中
        rsp->SetContent(body, "text/html");
    }
    // 把 HttpResponse 的要素按照 HTTP 格式进行组织发送
    void WriteResponse(const PtrConnection &conn, const HttpRequest &req, HttpResponse &rsp){
        // 1. 先完善头部字段
        if(req.Close()) rsp.SetHeader("Connection", "close");
        else rsp.SetHeader("Connection", "keep-alive");
        if(!rsp._body.empty()){
            if(!rsp.HasHeader("Content-Length")) rsp.SetHeader("Content-Length", std::to_string(rsp._body.size()));
            if(!rsp.HasHeader("Content-Type")) rsp.SetHeader("Content-Type", "application/octet-stream");
        }
        if(rsp._redirect_flag) rsp.SetHeader("Location", rsp._redirect_url);
        
        // 2. 将 rsp 中要素按照  http 协议格式进行组织
        std::stringstream rsp_str;
        rsp_str << req._version << " " << std::to_string(rsp._status) << " " << Util::StatuDesc(rsp._status) << "\r\n";
        for(auto &head: rsp._headers){
            rsp_str << head.first << ": " << head.second << "\r\n";
        }
        rsp_str << "\r\n";
        rsp_str << rsp._body;
        // 3. 发送数据
        conn->Send(rsp_str.str().c_str(), rsp_str.str().size());
    } 
    
    // 判断是否是静态资源
    bool IsFileHandler(const HttpRequest &req){
        // 1. 必须设置静态资源跟目录
        if(_basedir.empty()) return false;
        // 2. 请求方法, 必须是 GET/HEAD 请求方法
        if(req._method != "GET" && req._method != "HEAD") return false;
        // 3. 请求的资源路径必须是一个合法路径
        if(!Util::ValidPath(req._path)) return false;
        
        // 4. 请求资源必须存在, 且是一个普通文件
        // 注意: 有一种请求比较特殊 纯粹的目录: /, /image/, 这种情况给后边默认追加一个 index.html
        // index.html    /image/a.png
        // 不要忘了前缀的相对根目录,也就是将请求路径转换为实际存在的路径  /image/a.png  ->   ./wwwroot/image/a.png
        std::string req_path = _basedir + req._path;//为了避免直接修改请求的资源路径，因此定义一个临时对象 
        if(req._path.back() == '/') req_path += "index.html";
        if(!Util::IsRegular(req_path)) return false;

        return true;
    }

    // 静态资源请求处理
    void FileHandler(const HttpRequest &req, HttpResponse *rsp){
        std::string req_path = _basedir + req._path;
        if(req._path.back() == '/') req_path += "index.html";
        if(!Util::ReadFile(req_path, &rsp->_body)) return ;
        
        // 根据文件后缀名获取文件
        std::string mime = Util::ExtMime(req_path);
        rsp->SetHeader("Content-Type", mime);
        return;
    }
    // 功能性请求的分类处理
    void Dispatcher(HttpRequest &req, HttpResponse *rsp, Handlers &handlers){
        // 在对应请求方法的路由表中，查找是否含有对应资源请求的处理函数，有则调用，没有则发挥404
        // 思想：路由表存储的时键值对 -- 正则表达式 & 处理函数
        // 使用正则表达式，对请求的资源路径进行正则匹配，匹配成功就使用对应函数进行处理
        // /numbers/(\d+)  /numbers/12345
        for(auto &handler: handlers){
            // 两种区别
            // 每次请求都需编译构造出正则表达式对象, 但是这个过程比较慢, 因为每次请求都要编译, 因此映射表改成 regex, 而不是 string, 节省编译时间
            // const std::regex re(handler.first);
            const std::regex &re = handler.first;

            const Handler &func = handler.second;
            if(!std::regex_match(req._path, req._matches, re)) continue;
            // 转入请求信息 和 空的 rsp 执行处理函数
            return func(req, rsp);
        }
    }

    // 路由查找
    void Route(HttpRequest &req, HttpResponse *rsp){
        // 1. 分辨请求 -- 静态资源请求 | 功能性请求
        // GET、HEAD 默认认为是 静态资源请求
        if(IsFileHandler(req)) return FileHandler(req, rsp);
        if(req._method == "GET" || req._method == "HEAD") return Dispatcher(req, rsp, _get_route);
        else if(req._method == "POST") return Dispatcher(req, rsp, _post_route);
        else if(req._method == "PUT") return Dispatcher(req, rsp, _put_route);
        else if(req._method == "DELETE") return Dispatcher(req, rsp, _delete_route);
        
        rsp->_status = 405; // Method Not Allowed 请求方法不支持
        return ;
    }
    
    // 设置上下文
    void SetContext(const PtrConnection &conn){
        conn->SetContext(HttpContext());
        LOG_DEBUG("NEW CONNECTION %p", conn.get());
    }

    // 缓冲区数据解析 + 处理
    void ParseBuf(const PtrConnection &conn, Buffer *buffer){
        // 缓冲区内有数据持续处理
        while(buffer->ReadAbleSize() > 0){
            // 1. 获取上下文
            HttpContext *context = conn->GetContext()->get<HttpContext>();
            // 2. 通过上下文对缓冲区数据进行解析-- 得到 HttpRequest 对象
            context->RecvHttpRequest(buffer);
            HttpRequest &req = context->Request();
            HttpResponse rsp(context->RespStatus());
            if(context->RespStatus() >= 400){
                // 进行错误响应, 关闭连接
                ErrorHandler(req, &rsp);
                WriteResponse(conn, req, rsp); // 组织响应发送给客户端
                context->ReSet();
                // 出错时清除缓冲区数据 -- 在测试错误请求时会用到
                buffer->MoveReadOffset(buffer->ReadAbleSize()); 
                conn->Shutdown();//关闭连接
                return ;
            }
            if(context->RecvStatus() != RECV_HTTP_OVER) {
                // 当前请求还没有接收完整,则退出，等新数据到来再重新继续处理
                return;
            }
            // 3. 请求路由 + 业务处理 --> 得到一个填充了的 rsp 对象
            Route(req, &rsp);
            // 4. 对 rsp 对象进行组成成 HttpResponse 响应进行发送
            WriteResponse(conn, req, rsp);
            // 5. 重置上下文
            context->ReSet();
            // 6. 根据长短连接判断是否关闭连接 或者 继续处理
            if(rsp.Close()) conn->Shutdown();
        }
        return;
    }


public:
    HttpServer(int port, int timeout = DEFALT_TIMEOUT):_server(port){
        _server.EnableInactiveRelease(timeout);
        _server.SetConnectedCallback(std::bind(&HttpServer::SetContext, this, std::placeholders::_1));
        _server.SetMessageCallback(std::bind(&HttpServer::ParseBuf, this, std::placeholders::_1, std::placeholders::_2));
    }

    void Get(const std::string &pattern, const Handler &handler){
        _get_route.emplace_back(std::make_pair(std::regex(pattern), handler));
    }
    void Post(const std::string &pattern, const Handler &handler){
        _post_route.emplace_back(std::make_pair(std::regex(pattern), handler));
    }
    void Put(const std::string &pattern, const Handler &handler){
        _put_route.emplace_back(std::make_pair(std::regex(pattern), handler));
    }
    void Delete(const std::string &pattern, const Handler &handler){
        _delete_route.emplace_back(std::make_pair(std::regex(pattern), handler));
    }

    void SetBaseDir(const std::string &path){
        assert(Util::IsDirectory(path) == true);
        _basedir = path;
    }

    void SetThreadCount(int cnt){_server.SetThreadCount(cnt);}
    void EnableInactiveRelease(int timeout){_server.EnableInactiveRelease(timeout);}
    void Listen(){_server.Start();}
private:
    // 路由映射表
    Handlers _get_route;
    Handlers _post_route;
    Handlers _put_route;
    Handlers _delete_route;

    std::string _basedir; // 静态资源根目录
    TcpServer _server;
};
